export dataset_name=/nvme/tyang/workspace/Proteus/data/common_proteins

CUDA_VISIBLE_DEVICES=7 python src/VESPA/predict.py \
    --dataset_input $dataset_name \
    --prott5_cache /nvme/tyang/workspace/hestia/src/VESPA/models \
    --prott5cons_checkpoint /nvme/tyang/workspace/hestia/src/VESPA/models/ProtT5cons_checkpoint.pt \
